name: knowledge-grounding
endpoints:
- respond
compose:
  env_file:
  - .env
  build:
    args:
      SERVICE_PORT: 8083
      SERVICE_NAME: knowledge_grounding
      MODEL_CKPT: 3_sent_62_epochs
      DATA_ARG: http://files.deeppavlov.ai/alexaprize_data/parlai_grounding_knowledge/parlai_topical_chat_data.tar.gz
      MODEL1_ARG: http://files.deeppavlov.ai/alexaprize_data/parlai_grounding_knowledge/topical_chat_blender90_1_sent_48_epochs.tar.gz
      MODEL2_ARG: http://files.deeppavlov.ai/alexaprize_data/parlai_grounding_knowledge/topical_chat_blender90_3_sent_62_epochs.tar.gz
      CUDA_VISIBLE_DEVICES: '0'
      FLASK_APP: server
    context: ./services/knowledge_grounding/
  command: flask run -h 0.0.0.0 -p 8083
  environment:
  - CUDA_VISIBLE_DEVICES=0
  - FLASK_APP=server
  deploy:
    resources:
      limits:
        memory: 4G
      reservations:
        memory: 4G
  volumes:
  - ./services/knowledge_grounding:/src
  ports:
  - 8083:8083
